package org.example.demo3.demos.web.domain.dto;

import cn.hutool.core.bean.BeanUtil;
import cn.hutool.core.collection.CollUtil;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.baomidou.mybatisplus.extension.plugins.pagination.PageDTO;
import io.swagger.annotations.ApiModel;
import io.swagger.annotations.ApiModelProperty;
import lombok.Data;
import org.example.demo3.demos.web.User;

import java.util.Collections;
import java.util.List;
import java.util.function.Function;
import java.util.stream.Collectors;

@Data
@ApiModel(description = "分页结果封装")
public class  PageDto<T> {
    @ApiModelProperty("总条数")
    private Long total;
    @ApiModelProperty("总页数")
    private Long pageNos;
    @ApiModelProperty("列表")
    private List<T> list;

    public static <PO,VO> PageDto<VO> of(Page<PO> p, Function<PO,VO> convertor){ // 原本的类, 改成一个函数. 用来转化参数就可以. 不要直接copy了.

        PageDto<VO> dto = new PageDto<>();
        dto.setTotal(p.getTotal());
        dto.setPageNos(p.getPages());
        List<PO> records = p.getRecords();
        if(CollUtil.isEmpty(records)){
            dto.setList(Collections.emptyList());
            return  dto;
        }
        dto.setList(records.stream().map(convertor).collect(Collectors.toList())); // 把返回的数据转化为流,并且使用convertor函数映射. 最后收集一下就可以
        return  dto;
    }
    // 简单版本
 /*   public static <PO,VO> PageDto<VO> of(Page<PO> p,Class<VO> clazz){ // 指定范型POVO,指定返回值当前类的实例,类型是VO. of方法, 参数p是PO是实体类,Vo是返回类型,和dto很相似

        PageDto<VO> dto = new PageDto<>(); //  创建自己的实例
        dto.setTotal(p.getTotal());
        dto.setPageNos(p.getPages());
        List<PO> records = p.getRecords();
        if(CollUtil.isEmpty(records)){
            dto.setList(Collections.emptyList());
            return  dto;
        }
        dto.setList(BeanUtil.copyToList(records,clazz));
        return  dto;
    }*/
}
